package xin.marcher.module.quality.cor.impl;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import xin.marcher.module.quality.cor.RiskHandler;
import xin.marcher.module.quality.domain.dto.ItemQualityControlResultDTO;
import xin.marcher.module.quality.repository.ItemQualityControlRepository;

import java.util.Map;

/**
 *
 */
@Component
public class QualityWhitelistHandler implements RiskHandler {

    @Autowired
    private ItemQualityControlRepository itemQualityControlRepository;

    /**
     * 处理白名单中的风险数据，进入到这里的数据，就只需要对白名单进行判断，剔除掉白名单中的数据即可
     *
     * @param riskDataMap
     * @return
     */
    @Override
    public Map<String, ItemQualityControlResultDTO> handle(Map<String, ItemQualityControlResultDTO> riskDataMap) {
        // 白名单中的风险数据
        return itemQualityControlRepository.removeExistsInWhitelist(riskDataMap);
    }
}
